#include "../Base.h"



CommCallbackProc gCommCallback = NULL;
PULONG64 gWin7CommCallBack = NULL;
PULONG64 gWin10CommCallBack = NULL;




NTSTATUS RegisterCommucationCallBack(CommCallbackProc callback)
{

	RTL_OSVERSIONINFOW version = { 0 };
	RtlGetVersion(&version);
	if (version.dwBuildNumber == 7600 || version.dwBuildNumber == 7601) {
		return RegisterCommWin7(callback);
	}
	else {
		return RegisterCommWin10(callback);
	}
	

	return NTSTATUS();
}

VOID UnRegisterCommucationCallBack() {
	RTL_OSVERSIONINFOW version = { 0 };
	RtlGetVersion(&version);
	if (version.dwBuildNumber == 7600 || version.dwBuildNumber == 7601) {
		 UnRegisterCommWin7();
	}
	else {
		 UnRegisterCommWin10();
	}
}
